#ifndef DATABASE_H
#define DATABASE_H

#include "header.h"

#define DB_HOST "127.0.0.1"
#define DB_USER "root"


// 数据库连接管理
MYSQL* db_connect();  // 连接数据库
void db_close(MYSQL *conn);  // 关闭数据库连接

// 用户管理
int db_register_user(MYSQL *conn, const char *username, const char *password);  // 注册用户
int db_login_user(MYSQL *conn, const char *username, const char *password);  // 用户登录
int db_delete_user(MYSQL *conn, int user_id);  // 删除用户
int db_update_password(MYSQL *conn, int user_id, const char *new_password);  // 修改密码

// 消息管理
int db_save_message(MYSQL *conn, int sender_id, int receiver_id, const char *message, int status);  // 存储消息（支持已读/未读）
int db_update_message_status(MYSQL *conn, int message_id, int status);  // 更新消息状态（已读/未读）
int db_delete_message(MYSQL *conn, int message_id);  // 删除消息

// 聊天记录
void db_fetch_history(MYSQL *conn, int user1_id, int user2_id, int limit);  // 查询聊天记录，支持分页

#endif  // DATABASE_H


